"""Functions to load and write datasets."""

__all__ = [
    "load_airline",
    "load_arrow_head",
    "load_gunpoint",
    "load_basic_motions",
    "load_osuleaf",
    "load_italy_power_demand",
    "load_japanese_vowels",
    "load_plaid",
    "load_longley",
    "load_lynx",
    "load_shampoo_sales",
    "load_UCR_UEA_dataset",
    "load_unit_test",
    "load_uschange",
    "load_PBS_dataset",
    "load_japanese_vowels",
    "load_gun_point_segmentation",
    "load_electric_devices_segmentation",
    "load_acsf1",
    "load_macroeconomic",
    "load_hierarchical_sales_toydata",
    "generate_example_long_table",
    "load_from_arff_to_dataframe",
    "load_from_long_to_dataframe",
    "load_from_tsfile",
    "load_from_tsfile_to_dataframe",
    "load_from_ucr_tsv_to_dataframe",
    "make_multi_index_dataframe",
    "load_tsf_to_dataframe",
    "load_unit_test_tsf",
    "load_solar",
    "load_covid_3month",
    "load_forecastingdata",
    "load_m5",
    "write_panel_to_tsfile",
    "write_dataframe_to_tsfile",
    "write_ndarray_to_tsfile",
    "write_results_to_uea_format",
    "write_tabular_transformation_to_arff",
    "load_tecator",
    "load_fpp3",
    "_load_fpp3",
    "DATASET_NAMES_FPP3",
    "BaseDataset",
    "Airline",
    "Longley",
    "Lynx",
    "Macroeconomic",
    "PBS",
    "ShampooSales",
    "Solar",
    "USChange",
    "ForecastingData",
    "M5Dataset",
    "ACSF1",
    "ArrowHead",
    "BasicMotions",
    "GunPoint",
    "ItalyPowerDemand",
    "JapaneseVowels",
    "OSULeaf",
    "PLAID",
    "UCRUEADataset",
]

from sktime.datasets._data_io import (
    generate_example_long_table,
    make_multi_index_dataframe,
)
from sktime.datasets._fpp3_loaders import DATASET_NAMES_FPP3, _load_fpp3, load_fpp3
from sktime.datasets._hierarchical_demo import load_hierarchical_sales_toydata
from sktime.datasets._readers_writers.arff import (
    load_from_arff_to_dataframe,
    write_tabular_transformation_to_arff,
)
from sktime.datasets._readers_writers.long import load_from_long_to_dataframe
from sktime.datasets._readers_writers.ts import (
    load_from_tsfile,
    load_from_tsfile_to_dataframe,
    write_dataframe_to_tsfile,
    write_ndarray_to_tsfile,
    write_panel_to_tsfile,
)
from sktime.datasets._readers_writers.tsf import load_tsf_to_dataframe
from sktime.datasets._readers_writers.tsv import load_from_ucr_tsv_to_dataframe
from sktime.datasets._readers_writers.utils import write_results_to_uea_format
from sktime.datasets._single_problem_loaders import (
    load_acsf1,
    load_airline,
    load_arrow_head,
    load_basic_motions,
    load_covid_3month,
    load_electric_devices_segmentation,
    load_forecastingdata,
    load_gun_point_segmentation,
    load_gunpoint,
    load_italy_power_demand,
    load_japanese_vowels,
    load_longley,
    load_lynx,
    load_m5,
    load_macroeconomic,
    load_osuleaf,
    load_PBS_dataset,
    load_plaid,
    load_shampoo_sales,
    load_solar,
    load_tecator,
    load_UCR_UEA_dataset,
    load_unit_test,
    load_unit_test_tsf,
    load_uschange,
)
from sktime.datasets.base import BaseDataset
from sktime.datasets.classification import (
    ACSF1,
    PLAID,
    ArrowHead,
    BasicMotions,
    GunPoint,
    ItalyPowerDemand,
    JapaneseVowels,
    OSULeaf,
    UCRUEADataset,
)
from sktime.datasets.forecasting import (
    PBS,
    Airline,
    ForecastingData,
    Longley,
    Lynx,
    M5Dataset,
    Macroeconomic,
    ShampooSales,
    Solar,
    USChange,
)
